class OrderUI:
    def __init__(self, order_controller):
        self.controller = order_controller

    def add_order(self):
        product_name = input('请输入商品名称：')
        product_price = input('请输入商品价格：')
        product_count = input('请输入商品数量：')
        self.controller.add_order(product_name, product_price, product_count)

    def update_order_quantity(self):
        order_id = input('请输入订单ID：')
        new_quantity = input('请输入新的商品数量：')
        self.controller.update_order_quantity(order_id, new_quantity)

    def get_order(self):
        order_id = input('请输入订单ID：')
        self.controller.get_order(order_id)

    def order_menu(self):
        while True:
            print('1. 添加订单')
            print('2. 修改订单数量')
            print('3. 查询订单')
            print('4. 查询所有订单')
            print('5. 返回主界面')
            choice = input('请选择操作：')
            if choice == '1':
                self.add_order()
            elif choice == '2':
                self.update_order_quantity()
            elif choice == '3':
                self.get_order()
            elif choice == '4':
                self.controller.get_orders()
            else:
                print('返回主界面')
                break
